[feat] Added error checking to edit page