Reorganize screenshot functions