Exercise 4: Create Unit Tests. Whoops!
Update docs