Next step will be to include main function