fix bug, add mroe plots, stuff