Bug fixes + updates