I think that's the first assigment. This feels really hacky though