moved dropout to new name as now there is nn.Dropout (conflict of class names)