wip: better styling