Add unit spec and add more tests in lesson spec.