add results plots and improve generally
final version?