updated comments and readme