Cleaned up code. Adding screenshot and README.