Add documentation and make parameter more clear