Testing
Added a test to show that the functionality works.
Fixed code so it actually works now (oops!)