Just need to fix lag