Also, write the relevant unit test in the main function to test out these functionalities.
Also, write the relevant unit test in the main function to test out these functionalities.