added main class 

will work later tomorrow.