Adding a few features, updating readme and adding screenshots