Add more bug fixes