Add customizable workload parameter
Add function tests
Doc tests