Working now! Wrote a test class - it passes.