more styling, small bug fix