vairous bug fixes