Add pry
Fix bug