One more class was missing from repo, changed main