fix bugs 3,4 again
Fix all bugs