Added some styling - come back on the project after holidays