documenting code
adding TODOs; few minor changes
updating readme