Tests passing for first stage