Added tests for editing plans