Adds styling and removes extra code