Added functionality to edit problems.