Fixing ugly code. Add product button still not working.