Fixed up on some styling