As suggested in #10 . I have made the changes . Please review.