added TODO and improved readme