add dropout into code