Add initialization logic