Changed DatabaseTest to if else format.