added pythonic way of if statement with or