Refactored and added header.rb