add new param
fixing codes #1
fixing codes #2