Clean up readme and add screenshots