edit to assignment3 models