Fix some issues
Merge pull request #25 from dian4554/lb-params

Fix some issues