Fixes, improvements til task 6