added more code

to main