Bug fixes galore