Added another if-statement