minor edits, removed unecessary imports