Added some cmd line input for testing