Added the ability to choose the color of the new shape
Fixed small bug