minor edits
bug fixed