Fixed and expanded the existing tests