Bug fixes + debug