Fixing up the UI to make things a bit more clear
Adding more to the readme