Updated README with instructions on how to run code + TODO list