[IMPROVE] enable gpu build with energy support

Signed-off-by: Anastasia Lyupa <anastasiya.lyupa@gmail.com>
[IMPROVE] move data init to main, unset wells

Signed-off-by: Anastasia Lyupa <anastasiya.lyupa@gmail.com>
[IMPROVE] output plots names

Signed-off-by: Anastasia Lyupa <anastasiya.lyupa@gmail.com>