add screenshot and refactor demo