Added some comments and a little bit more styling