Added a class and updated style