Basic styling done. Need refactoring