Updated assignment starter code