Bug fixes + api updates