fixed some smaller issues with heatmaps
plotting layered graphs now works for multilayer dbns