Adding def main()