Working: added test unit
Working: added test unit