This is the version for the assignment