minor bug fix
minor edits
done