more updates on code and readme