Forgot about input sizes