I feel like you can do better than this. Loop?