add better styling