Added another condition and tested

Added another condition and tested successfully