method works in irb but tests not passing. Come back to fix tests