bug fixes add comment