improved comments and updated readme