Add linear gradient tests
Clenaup header file inclusion