improved UI/UX and using .val() correctly