Bug fixes galor...