add start of visualization
improvements to gbm and better testing