some better styling clean up