Added elif statement for readability