changes in code (+ mini bug fix in report)