Add unit tests for aaron