Adding check to ensure args[1] is integer.