Added tests for initialize function