Fixed up more of the styling ready to invest myself to this design