Adding an elif test
Adding single-line test