Enchance readme and add more comments