cleaning up documentation and code