Working, except input gains argument