Added Edit class