Bug fix on functions, added arg function.