Test

Add test unit for sort function