Bug fixes galour!