Dropout implementatin is simplified