update Main file for better