Add bug fixes