Updated .h file