Added tests for all stages.