Add initialization step