Made changes to main.swift sample code