(chore) #4 Update documentation and clean up code.