Remove unecessary code