Added more junit tests for head