1 files changed, 13 insertions, 8 deletions

diff --git a/src/srcut.cpp b/src/srcut.cpp
index 6df435b..0e5654a 100644
--- a/src/srcut.cpp
+++ b/src/srcut.cpp
@@ -83,26 +83,31 @@ operator>>(istream& in, map<string, vector<string>>& container)
 }
 
 ostream&
-operator<<(ostream& os, const map<string, vector<string>>& container)
+print(ostream& os, const string& key, const map<string, vector<string>>& container)
 {
-        for ( const auto& chunk: container ) {
-                os << "CHUNK: " << chunk.first << '\n';
-                auto line = begin(chunk.second);
-                while ( line != end(chunk.second) ) {
+        const auto it = container.find(key);
+        if ( it != container.end() ) {
+                auto line = begin(it->second);
+                while ( line != end(it->second) ) {
                         os << *line << '\n';
 
                         ++line;
                 }
-                os << "-------\n\n";
+        }
+        else {
+                os << "ERROR: " << key << " not found!" << endl;
         }
         return os;
 }
 
 int
-main(void)
+main(int argc, char* argv[])
 {
         map<string, vector<string>> container;
 
         cin >> container;
-        cout << container;
+
+        for ( int i = 1; i < argc; ++i ) {
+                print(cout, argv[i], container);
+        }
 }