Might need a refactor before downloading file...