Revised drag function