almost done except one problem