Updated first docstring of homework 5