Almost done
Only challenging problem todo