Add flag parsing