changed the main file a little