Proper code edit