Refactor
Support the --only flag

Todo: --extra-funs flag