bug fixes ans additions