Add bugfixes