fix a couple more bugs
updates.