Updated to be same as prof's example.