Tried to add dropout