Working on optimizing performance
Fixed a few issues where the squre wouldn't be selected and added a selected shape to increase performance