added support for small and medium configurations