Add initialization part