Problem solver using Swift