Started work on unit testing.