Added all tests - refactored text into its own class however test doesn't work