Bug fix
using class instead of id for bigscreen & smallscreen