Add more weights and change input method