Added another if statement