Working example kinda

Issue with heights still