Tidy EditData code