rename method and move context stuff outside of it
Make lines look better