second commit. Main function added