A bit off with the glados test cases but close I guess!