Big fixes, test updates