Add imports
get arguments
refactor and add solving