Updated readme with screenshot and TODO