Updated readme with TODO and screenshots