cleanup, add inputs

Signed-off-by: FeiGuan <guanfei12345@sina.com>