Created own header and changed .cpp to work with it