Updates main class due to git weirdness