Refactor, update readme, add screenshot