Add Initial weights initialisation