I've got it working now, just need to fine tune layout