Made it recognize three of a kind and added more tests