Did some more changes to the styling