Suggestion for adjustment

Hey, I've made some adjustments regarding my comments in #5, they're just suggestions so do as you please. :)