Added some comments, nice indentation