Fix leaf gain function
Adds cli parameters
Parse node id from dot file
get testing set gain