Feature/improving tests for flashcard