Added parameters to main

Added parameters to main