final stage. Added command line argument