minor bug fix and styling changes