Forgot to pass in arguments! Passing tests, let's try to refactor this.