Adding Header Files as Jaylene suggested