Add function for saving user problem