Renamed the code, got it working completely