Added test to confirm changes work