So I think I have the right idea but it's not working :( need to debug