bugs found after testing

fixed