Add code docs
Remove unnecessary cruft