Wraps logic in an if, for now