Renamed model names to match the names in the paper