refactor a little
Mark down