Final bug fixes