Add features inspired by Tania's GNN

Created by: asogaard

See https://drive.google.com/drive/folders/1JlYFywgUH0kupz7XMOkip3gdhGro5QzA, slide 8