added elif statement