Added lines to initialize

Hi! Added to the initialize part