Versatility

Changed a little bit of code for simplicity’s sake.