updating css, cleaning up code for this step