add some styling back in