edit to assignment 2