Add pry for test