Bug fixes cont...