Add argument handling for the following: - [ ] create statistics - [ ] train - [ ] validation - [ ] testing