part 2 done
fix