Added final code

Added a formal test function and added user input to create any tree of
any height.