Refactored run method to return just $body for callback, updated relevant tests