Got first test kinda working