Yeah, sure! I would maybe add an indicator on the brush showing which color is currently selected. Adding sound effects while painting would also make it feel more satisfying. It would also be cool to have different sfx/vfx based on the effect the paintings have on you.
