got asssignment 4 functioning, not working