Small bugfix with adding and removing squares