added prompt to get weight