bug fixes & updates