Clean up example, add comments

- Currently comments are discarded