Adding possibility to get beta vector