* Fixed up changes to some of the content
* Fixed some imports
* Removed some extra text
* Polished up a bit more stuff and fixed a portion about the error message
* Apply suggestions from code review
Co-authored-by: Jason Liu <jxnl@users.noreply.github.com>
---------
Co-authored-by: Jason Liu <jxnl@users.noreply.github.com>