working on more styling