minor changes
num grad added
minor changes
minor changes
minor changes