Add some stuff in unit test
changes in unittest1