Bug fixes galore!