Removing unessecary code