Adding assertions + testing.