one line issue fixed and tests added