Added documentation in code :)