Removed extra code. Now just have to add shapes