- [ ] Remove extraneous comments - [ ] Add more documentation about input data structure and different figure types