Added stratified sampmling, updated readme and added spacebar to save image