Took couple steps back. Got rid of unnecessary code