edited for create class
add Main