fixed bug so all cases work now
removed pry