Refactor to make it work