Updated to Julia Wilson's suggestion for class change.
JSHint passing!