bug fixes -2

bug fixes -2