Changed edit_category over to wtforms