Made a method simpler