Added functionality to choice 3