Added basic documentation
Improved comments