code refactoring
readme update