make it modular done, had to check answer