Re-wrote example tests