Modify edit to new class structure