Remove magic numbers from main