Added main class and reference class