still frustrated with if statements, made it more simple for now