Added some comments and worked on size