add new feature (continue #4)

Signed-off-by: rei <reiheart@gmail.com>