Put 2 codes together! (School project)