Removed binding.pry from the code