Added functionality to pick a problem