Fixed issue with jquery & a bit of styling