Added more Tests

However, either diff isn't working or my check for diff is incorrect.