Updated solution
modified header