Added testing via the __name__ == __main__