Edit main function, edit readme