Cleaned interface and code

Added some comments
Added print tree feature