Add requirements
Set up return type for docstring