Add workload to job. Closes #38
Fix tests