made the code a little more user friendly