Added a test to prove the failing code. Still have to figure out how to fix