Updated the run classification method