added some styling and some logic