Remove unneccessary comments. Add docstrings