Adding styling, close #2