Optimisation of a simple equivalent dipole model to include edge effects