added some more code

just need to verify some code