updated code-first homework