Added some extra styling and started thinking about js