changes to calling class, can now input difficulty