small bug fix and formatting