Forgot about actually making it work, and small fix.