Trying to fix int() problem