Working on unit testing.