Simplified tests.
Simplified main func.