Add some more logic