CHANGES dropout model to simpler version