Add code to handle file upload