Missing newline at end of file. Also please add docstring.