refactor (im still learning python ok!?!?!)