Use 'if' statement to pass first test