Just need to make it work now