Main function added