Removed square code because I couldnt get that working yet.