bug fixes galore