Put the original test code in.

Original test code from the course website.