Added paint choice functionality