Added initialization to init all the layers