updated .h files - WIP