Federal budget to focus on gender equality