final bug fixes