Added args test. Fixed code style error. Fixed missing import bug in unit test.