fix issue #9
prettying up