Simple edits

made things more clean