almost done. need to fix one more thing