Updated comments and readme