Fixing up a bug in the original code - getting tests to pass