Removed unncessary code and div