Update main

added iostream library