modified main file a bit