Added some styling after Jeremie's feedback