Futher bug fixes