Adjusted things to work with new layout