minor bug fix and styling