Added some styles / wip