a few tweaks. working on HW1