bug fixes latest code